import akshare as ak
import pandas as pd
import backtrader as bt
import datetime
# 利用 AKShare 获取股票的后复权数据，这里只获取前 6 列
def get_data(code, start_date, end_date):
    stock_hfq_df = ak.stock_zh_a_hist(symbol=code, adjust="qfq", start_date=start_date, end_date=end_date).iloc[:, :6]
    # 处理字段命名，以符合 Backtrader 的要求
    stock_hfq_df.columns = [
    'date',
    'open',
    'close',
    'high',
    'low',
    'volume',
    ]
    # 把 date 作为日期索引，以符合 Backtrader 的要求
    stock_hfq_df.index = pd.to_datetime(stock_hfq_df['date'])
    return bt.feeds.PandasData(dataname=stock_hfq_df, 
                           fromdate=start_date, 
                           todate=end_date)  # 加载数据